#include "iapmergedialog.h"
#include "ui_iapmergedialog.h"

IAPMergeDialog::IAPMergeDialog(QWidget *parent)
    : QDialog(parent)
    , ui(new Ui::IAPMergeDialog)
{
    ui->setupUi(this);

    setWindowTitle("固件合并工具");
    setWindowFlag(Qt::WindowContextHelpButtonHint, false);  // 去除标题栏的?
    setModal(true);

    initTableHeader();

    QObject::connect(ui->btnAddFile,
                     static_cast<void (QPushButton::*)(bool)>(&QPushButton::clicked),
                     this,
                     static_cast<void (IAPMergeDialog::*)(void)>(&IAPMergeDialog::addFile));
    QObject::connect(ui->btnDelFile,
                     static_cast<void (QPushButton::*)(bool)>(&QPushButton::clicked),
                     this,
                     static_cast<void (IAPMergeDialog::*)(void)>(&IAPMergeDialog::delFile));
    QObject::connect(ui->btnGenerateFile,
                     static_cast<void (QPushButton::*)(bool)>(&QPushButton::clicked),
                     this,
                     static_cast<void (IAPMergeDialog::*)(void)>(&IAPMergeDialog::generateFile));
}

IAPMergeDialog::~IAPMergeDialog()
{
    delete ui;
}

void IAPMergeDialog::msgBoxError(QString text)
{
    QMessageBox::critical(this, QApplication::applicationName(), text);
}

void IAPMergeDialog::msgBoxInfo(QString text)
{
    QMessageBox::information(this, QApplication::applicationName(), text);
}

void IAPMergeDialog::msgBoxWarn(QString text)
{
    QMessageBox::warning(this, QApplication::applicationName(), text);
}

void IAPMergeDialog::initTableHeader()
{
    if(!model) {
        model = new QStandardItemModel(0, 5, this);
    }

    QStringList headers;
    headers << "文件名称"
            << "起始地址(HEX)"
            << "结束地址(HEX)"
            << "占用空间(字节)"
            << "校验码(CRC32)";
    model->clear();
    model->setHorizontalHeaderLabels(headers);
    ui->tableView->setModel(model);
    ui->tableView->setEditTriggers(QTableView::NoEditTriggers);           // 不可编辑
    ui->tableView->setSelectionBehavior(QAbstractItemView::SelectRows);   // 行选中
    ui->tableView->setSelectionMode(QAbstractItemView::SingleSelection);   // 单选
    ui->tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);  // 拉伸填满
    ui->tableView->horizontalHeader()->setSectionResizeMode(0, QHeaderView::Interactive); // 第一列可拉伸
    ui->tableView->horizontalHeader()->resizeSection(0, 170);  // 设置第一列宽度
}

void IAPMergeDialog::addFile()
{
    if(lastSavePath.isEmpty()){
        lastSavePath = "./";
    }
    QString filename = QFileDialog::getOpenFileName(this,
                                                    "选择固件",
                                                    lastSavePath,
                                                    "程序文件 (*.bin *.hex);;程序文件 (*.bin);;程序文件 (*.hex)");
    if(filename.isEmpty()){
        return;
    }

    foreach(endpoint item, fileList){
        if(item.filename == filename){
            QString text = QString("错误：文件已经选择！");
            msgBoxError(text);
            return;
        }
    }
    QByteArray data;
    uint32_t flashBaseAddress = 0;
    QFileInfo fileInfo(filename);
    QFile file(filename);
    if(!file.open(QIODevice::ReadOnly)){
        msgBoxError("错误：文件打开失败，请检查该文件是否有效！");
        return;
    }

    if(fileInfo.suffix().toLower() == "bin")  // 选择的是bin文件，那么需要填写FLASH地址
    {
        InputAddressDialog dialog(this);
        if(dialog.exec() == QDialog::Rejected){
            return;
        }
        flashBaseAddress = dialog.getAddress();
        data = file.readAll();
    }else if(fileInfo.suffix().toLower() == "hex")
    {
        int rows = 0;
        while(true)
        {
            QByteArray line = file.readLine();
            if(line.isEmpty()){
                break;
            }

            if(line.size() < 12 || line[0] != ':' || line[line.size() - 1] != '\n')
            {
                msgBoxError("错误：错误的HEX文件内容！");
                return;
            }

            line.remove(0, 1);  // 去除冒号
            line = line.simplified();  // 去除回车换行
            line = QByteArray::fromHex(line);  // 转十六进制

            // 累加和校验
            char checksum = 0;
            for(int i = 0; i < line.size() - 1; i++)
            {
                checksum += line[i];
            }
            checksum = 0x100 - checksum;
            if(checksum != line[line.size() - 1])
            {
                msgBoxError("错误：错误的HEX文件内容！");
                return;
            }

            // 记录文件起始FLASH地址
            if(rows == 0)
            {
                if(line[0] != (char)0x02 || line[3] != (char)0x04)
                {
                    msgBoxError("错误：错误的HEX文件内容！");
                    return;
                }
                flashBaseAddress = ((line[4] << 8) | line[5]) << 16;
            }
            if(rows == 1)
            {
                if(line[3] != (char)0x00)
                {
                    msgBoxError("错误：错误的HEX文件内容！");
                    return;
                }
                flashBaseAddress |= ((line[1] << 8) | line[2]);
            }

            // 根据类型解析数据
            if((uint8_t)line[3] == 0) // 数据
            {
                data += QByteArray(line.data() + 4, line[0]);
            }else if((uint8_t)line[3] == 1) // 结束
            {
                break;
            }if((uint8_t)line[3] == 4) // 地址
            {
                ;
            }
            rows += 1;
        }
#if 0
        {
            QFile f("comp.bin");
            f.open(QIODevice::ReadWrite | QIODevice::Truncate);
            f.write(data);
            f.close();
        }
#endif
    }else
    {
        msgBoxError("错误：文件格式无效！");
        return;
    }

    foreach(endpoint item, fileList){
        if(flashBaseAddress >= item.flashBeginAddress && flashBaseAddress <= item.flashEndAddress){
            QFileInfo info(item.filename);
            QString text = QString("错误：地址存在重叠区域，请重新核对！\r\n重叠文件为：%1。").arg(info.fileName());
            msgBoxError(text);
            file.close();
            return;
        }
    }

    // 记录文件列表
    endpoint ep;
    ep.filename = fileInfo.fileName();
    ep.filepath = fileInfo.filePath();
    ep.flashBeginAddress = flashBaseAddress;
    ep.flashEndAddress = flashBaseAddress + data.size();
    ep.data = data;
    ep.crc = CRC32(data.data(), data.size());
    fileList.append(ep);

    // 记录当前文件名的路径
    lastSavePath = fileInfo.path();

    // 根据地址FLASH起始地址重新排序
    std::sort(fileList.begin(), fileList.end(), [](const endpoint &a,const endpoint &b) {
        return a.flashBeginAddress < b.flashBeginAddress;
    });

    // 重新显示表格
    initTableHeader();

    foreach(endpoint ep, fileList)
    {
        QList<QStandardItem *> row;
        // 文件名称
        row.append(new QStandardItem(ep.filename));
        // 起始地址
        row.append(new QStandardItem(QString::asprintf("0x%.8X", ep.flashBeginAddress)));
        // 结束地址
        row.append(new QStandardItem(QString::asprintf("0x%.8X", ep.flashEndAddress)));
        // 占用空间
        row.append(new QStandardItem(QString::asprintf("%d", ep.data.size())));
        // CRC32
        row.append(new QStandardItem(QString::asprintf("0x%.8X", ep.crc)));
        // 设置表格显示
        for (int i = 0; i < row.size(); i++) {
            row[i]->setTextAlignment(Qt::AlignCenter);  // 居中
        }
        model->appendRow(row);
    }
}

void IAPMergeDialog::delFile()
{
    int row = ui->tableView->currentIndex().row();
    if(row < 0)
    {
        return;
    }
    fileList.removeAt(row);
    model->removeRow(row);
}

void IAPMergeDialog::generateFile()
{
    if(fileList.isEmpty()){
        msgBoxError("请先添加固件文件！");
        return;
    }

    // 计算总固件长度
    QFileInfo lastFileInfo(fileList.last().filename);
    uint32_t size = fileList.last().flashEndAddress - fileList.first().flashBeginAddress;

    // 合并固件
    QByteArray data(size, 0);
    for(int i = 0; i < fileList.length(); i++)
    {
        uint32_t offset = fileList[i].flashBeginAddress - FLASH_ADDRESS_BEGIN;
        memcpy(data.data() + offset, fileList[i].data.data(), fileList[i].data.size());
    }

    // 保存固件
    QString filename = QFileDialog::getSaveFileName(this, "保存文件", lastSavePath, "*.bin");
    if(filename.isEmpty()){
        return;
    }
    QFile file(filename);
    if(!file.open(QIODevice::ReadWrite | QIODevice::Truncate)){
        msgBoxError("错误：保存失败，请重试！");
        file.close();
        return;
    }
    if(file.write(data) != data.length()){
        msgBoxError("错误：保存失败，请重试！");
        file.close();
        return;
    }
    file.close();
    msgBoxInfo(QString("远程文件【%1】已保存！").arg(QFileInfo(filename).fileName()));

    // 打开输出目录并定位到固件文件
    QProcess process(this);
    QString cmd(tr("explorer /select, \"%1\"").arg(filename.replace("/", "\\")));
    process.startDetached(cmd);
}

unsigned int IAPMergeDialog::CRC32(const void *buf, unsigned int len)
{
    static const unsigned int crc32tab[] =
    { 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419,
      0x706af48f, 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4,
      0xe0d5e91e, 0x97d2d988, 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07,
      0x90bf1d91, 0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
      0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7, 0x136c9856,
      0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
      0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4,
      0xa2677172, 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
      0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3,
      0x45df5c75, 0xdcd60dcf, 0xabd13d59, 0x26d930ac, 0x51de003a,
      0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423, 0xcfba9599,
      0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
      0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190,
      0x01db7106, 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f,
      0x9fbfe4a5, 0xe8b8d433, 0x7807c9a2, 0x0f00f934, 0x9609a88e,
      0xe10e9818, 0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
      0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e, 0x6c0695ed,
      0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
      0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3,
      0xfbd44c65, 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
      0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a,
      0x346ed9fc, 0xad678846, 0xda60b8d0, 0x44042d73, 0x33031de5,
      0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa, 0xbe0b1010,
      0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
      0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17,
      0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6,
      0x03b6e20c, 0x74b1d29a, 0xead54739, 0x9dd277af, 0x04db2615,
      0x73dc1683, 0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
      0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1, 0xf00f9344,
      0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
      0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a,
      0x67dd4acc, 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
      0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1,
      0xa6bc5767, 0x3fb506dd, 0x48b2364b, 0xd80d2bda, 0xaf0a1b4c,
      0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55, 0x316e8eef,
      0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
      0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe,
      0xb2bd0b28, 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31,
      0x2cd99e8b, 0x5bdeae1d, 0x9b64c2b0, 0xec63f226, 0x756aa39c,
      0x026d930a, 0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
      0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38, 0x92d28e9b,
      0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
      0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1,
      0x18b74777, 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
      0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45, 0xa00ae278,
      0xd70dd2ee, 0x4e048354, 0x3903b3c2, 0xa7672661, 0xd06016f7,
      0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc, 0x40df0b66,
      0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
      0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605,
      0xcdd70693, 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8,
      0x5d681b02, 0x2a6f2b94, 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b,
      0x2d02ef8d};
    const unsigned char *p = (const unsigned char*) buf;
    unsigned int crc = 0xFFFFFFFF;
    while (len--)
    {
        crc = crc32tab[(crc ^ *p) & 0xff] ^ (crc >> 8);
        p++;
    }
    return crc ^ 0xFFFFFFFF;
}
